hackerrank array mathematics problem solution in python programming